//tcp_client.cc
#include <iostream>
#include <sys/socket.h>
#include <sys/types.h>
#include <string>
#include <cerrno>
#include <cstring>
#include <netinet/in.h>
#include <arpa/inet.h>
#include "Err.hpp"
#include <unistd.h>
#include <functional>
using namespace std;
//用来处理服务器启动方法错误
void Usage(const string& proc)
{
    cout << "Method of Use:"<<proc<<" clientip clientport" << endl;
}
int main(int argc, char *argv[])
{
    //保证不会启动错误
    if (argc != 3)
    {
        Usage(argv[0]);
        exit(USE_ERR);
    }

    //1.创建套接字
    //SOCK_STREAM是面向连接、可靠的数据传输服务
    int sock = socket(AF_INET, SOCK_STREAM, 0);
    if (sock < 0)
    {
        cout << "socket error" << endl;
        exit(SOCKET_ERR);
    }

    //2.需要bind，但不需要自己bind
    uint16_t serverport = atoi(argv[2]);
    string serverip = argv[1];

    //放入结构体中，用于后面找到服务器通信
    struct sockaddr_in server;
    //将server空间全部设置为\0
    bzero(&server, sizeof(server));
    //// 将服务器的IPv4点十进制IP的字符串转换为IN_ADDR结构的二进制IP
    server.sin_addr.s_addr = inet_addr(serverip.c_str());
    //AF_INET表示是IPv4通信类别
    server.sin_family = AF_INET;
    //将服务器的端口号转换成网络序列
    server.sin_port=htons(serverport);
    int cnt = 5;

    //3.进行连接
    //connect返回表示连接成功
    while (0 != connect(sock, (struct sockaddr *)&server, sizeof(server)))
    {
        sleep(1);
        cerr << "正在重新链接，重连次数还有：" << cnt-- << endl;
        if (cnt <= 0)
        {
            break;
        }
    }
    if (cnt == 0)
    {
        cerr << "链接失败" << endl;
        exit(CONNECT_ERR);
    }


    //4.连接成功后处理业务
    while (1)
    {
        cout << "input message:";
        string message;
        //从终端获取一段信息
        getline(cin, message);
        //向服务器进行发送信息
        write(sock, message.c_str(), message.size());

        //获取服务器响应的信息
        char buff[1024];
        ssize_t n = read(sock, buff, sizeof(buff) - 1);
        if (n > 0)
        {
            buff[n] = 0;
            cout << "server echo:" << buff << endl;
        }
        else if (n == 0)
        {
            // 对方把链接关闭了
            cerr << "server quit" << endl;
            break;
        }
        else
        {
            cerr << "read error" << strerror(errno) << endl;
            break;
        }
    }
    return 0;
}
